How to write a program of simple interest using function in python